package com.adc.douyu_assistant.service.impl;

import com.adc.douyu_assistant.entity.DanmuMessage;
import com.adc.douyu_assistant.repository.DanmuMessageRepository;
import com.adc.douyu_assistant.service.DanmuMessageService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * @Description
 * @Author lilin
 * @Create 2018-09-14 15:11
 **/
@Service
public class DanmuMessageServiceImpl implements DanmuMessageService {
    @Autowired
    private DanmuMessageRepository danmuMessageRepository;

    @Override
    public void save(List<DanmuMessage> danmuMessages) {
        danmuMessageRepository.saveAll(danmuMessages);
    }

    @Override
    public void save(Map<String, DanmuMessage> danmuMessageMap) {
        List<DanmuMessage> danmuMessages = new ArrayList<>();
        for (String key : danmuMessageMap.keySet()) {
            danmuMessages.add(danmuMessageMap.get(key));
        }
        danmuMessageRepository.saveAll(danmuMessages);
        System.out.println("save to DB success");
    }
}
